import os
from multiprocessing import Pool

def sur_cmds_exp_1_6_1():
    all_cmds = []

    seeds = range(2)
    model_types = ['TranDGN-tran']
    difficulty_levels = ("easy", "hard", "medium")
    for seed in seeds:
        for model in model_types:
            for edl in difficulty_levels:
                cmd = f'python main.py -env Surviving -edl {edl} -m {model} -S {seed}'
                all_cmds.append(cmd)

    return all_cmds

if __name__ == "__main__":
    paral_num = 1
    cmds = sur_cmds_exp_1_6_1()
    print(f"will execute {len(cmds)} commands:", *cmds, sep="\n")
    pool = Pool(processes=paral_num)
    pool.map_async(os.system, cmds)
    pool.close()
    pool.join()